package com.demo.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.BooleanUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.demo.dto.Result;
import com.demo.entity.Shop;
import com.demo.mapper.TbShopMapper;
import com.demo.service.IShopService;
import com.demo.utils.CacheClient;
import com.demo.utils.RedisData;
import jakarta.annotation.Resource;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalDateTime;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;

import static com.demo.utils.RedisConstants.*;

/**
* @author 小雷
* @description 针对表【tb_shop】的数据库操作Service实现
* @createDate 2025-10-04 14:34:52
*/
@Service
public class ShopServiceImpl extends ServiceImpl<TbShopMapper, Shop>
    implements IShopService {

    @Resource
    private StringRedisTemplate stringRedisTemplate;

    @Resource
    private CacheClient cacheClient;

    @Override
    public Result queryById(Long id) {

        Shop shop = null;

        // 解决缓存穿透
        // shop = queryWithPassThrough(id);

        // 互斥锁解决缓存击穿
        // shop = queryWithMutex(id);

        // 逻辑过期解决缓存击穿
        // shop = queryWithLogicalExpire(id);

        // 调用工具类解决缓存穿透
        shop = cacheClient.queryWithPassThrough(CACHE_SHOP_KEY, id,
                Shop.class, this::getById, CACHE_SHOP_TTL, TimeUnit.MINUTES);

        // 调用工具类, 使用互斥锁解决缓存击穿
//        shop = cacheClient.queryWithMutex(CACHE_SHOP_KEY, LOCK_SHOP_KEY, id,
//                Shop.class, this::getById, CACHE_SHOP_TTL, TimeUnit.MINUTES);

        // 调用工具类, 使用逻辑过期, 解决缓存击穿
//        shop = cacheClient.queryWithLogicalExpire(CACHE_SHOP_KEY, id,
//                Shop.class, this::getById, CACHE_SHOP_TTL, TimeUnit.MINUTES);

        // 返回友好的信息给客户端
        if(shop == null){
            return Result.fail("店铺不存在!");
        }
        // 返回商铺信息
        return Result.ok(shop);
    }


    @Override
    @Transactional
    public Result updateShop(Shop shop) {
        // 获取店铺 ID
        Long id = shop.getId();
        // ID 为空, 店铺不存在
        if(id == null){
            return Result.fail("更新失败, 店铺不存在!");
        }

        // ID 不为空, 更新数据库
        updateById(shop);
        // 删除缓存
        stringRedisTemplate.delete(CACHE_SHOP_KEY + id);
        return Result.ok();
    }

    /**
     * 获取互斥锁, 用于解决缓存击穿
     * @param key
     * @return
     */
    private boolean getLock(String key){
        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent(key, "1", LOCK_SHOP_TTL, TimeUnit.SECONDS);
        return BooleanUtil.isTrue(flag);
    }

    /**
     * 释放互斥锁
     * @param key
     */
    private void delLock(String key){
        stringRedisTemplate.delete(key);
    }

    /**
     * 解决缓存穿透
     * @param id
     * @return
     */
    private Shop queryWithPassThrough(Long id){
        // key
        String key = CACHE_SHOP_KEY + id;

        // 判断缓存是否命中
        String shopJson = stringRedisTemplate.opsForValue().get(key);

        // 缓存命中, 并且缓存数据不是空值
        if(StrUtil.isNotBlank(shopJson)){
            return JSONUtil.toBean(shopJson, Shop.class);
        }

        // 缓存命中, 并且缓存数据是空值（空字符串表示店铺不存在）
        if(shopJson != null && shopJson.equals("")){
            // 返回错误信息
            return null;
        }

        // 缓存未命中, 查询数据库
        Shop shop = getById(id);

        // 商铺不存在
        if(shop == null){
            // 将空值写入 Redis（使用特殊标记表示空值）
            stringRedisTemplate.opsForValue().set(key, "", CACHE_NULL_TTL, TimeUnit.MINUTES);

            // 返回错误信息
            return null;
        }

        // 商铺存在, 写入 Redis
        stringRedisTemplate.opsForValue().set(key, JSONUtil.toJsonStr(shop),CACHE_SHOP_TTL, TimeUnit.MINUTES);

        return shop;
    }

    /**
     * 基于互斥锁解决缓存击穿
     * @param id
     * @return
     */
    private Shop queryWithMutex(Long id){
        // 设置 key
        String key = CACHE_SHOP_KEY + id;

        // 查询缓存
        String shopJson = stringRedisTemplate.opsForValue().get(key);

        // 命中缓存 (当前不存在缓存穿透问题, 查到的结果一定不为空值)
        if(StrUtil.isNotBlank(shopJson)){
            return BeanUtil.toBean(shopJson, Shop.class);
        }

        // 命中空值
        if(shopJson != null){
            return null;
        }

        // 未命中缓存, 使用互斥锁重建缓存
        // 构造互斥锁的 key, 而不是商铺的 key
        String lockKey = LOCK_SHOP_KEY + id;
        Shop shop = null;

        try {

            // 获取互斥锁
            boolean isLock = this.getLock(lockKey);

            // 获取失败
            if(!isLock){
                // 当前线程休眠, 然后重新获取互斥锁
                Thread.sleep(50);
                queryWithMutex(id);
            }
            // 获取成功, 再次查询缓存, 做 double check
            shopJson = stringRedisTemplate.opsForValue().get(key);

            if(StrUtil.isNotBlank(shopJson)){
                return BeanUtil.toBean(shopJson, Shop.class);
            }

            // 查询数据库
            shop = getById(id);

            // 模拟缓存重建
            Thread.sleep(200);

            // 数据库中没有数据, 触发缓存击穿, 存入空值到 redis
            if(shop == null){
                stringRedisTemplate.opsForValue().set(key, "", CACHE_SHOP_TTL, TimeUnit.MINUTES);
                return null;
            }

            // 写入缓存
            stringRedisTemplate.opsForValue().set(key, JSONUtil.toJsonStr(shop), CACHE_SHOP_TTL, TimeUnit.MINUTES);

        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        } finally {
            // 释放锁
            this.delLock(lockKey);
        }

        // 返回商铺信息
        return shop;
    }

    /**
     * 存店铺信息到缓存, 并设置逻辑过期时间
     * @param id 店铺 id
     * @param expireSeconds 逻辑过期时间
     */
    public void saveShop2Redis(Long id, Long expireSeconds){
        // 查询店铺数据
        Shop shop = getById(id);

        // 实例 RedisData 并设置属性
        RedisData redisData = new RedisData();
        redisData.setData(shop);
        // 在本地时间上 + 多少秒
        redisData.setExpireTime(LocalDateTime.now().plusSeconds(expireSeconds));

        // 写入 Redis, 不设置 TTL, 而是使用逻辑过期
        stringRedisTemplate.opsForValue().set(CACHE_SHOP_KEY + id, JSONUtil.toJsonStr(redisData));
    }

    private static final ExecutorService CACHE_REBUILD_EXECUTOR = Executors.newFixedThreadPool(10);

    /**
     * 使用逻辑过期解决缓存击穿
     * @param id
     * @return
     */
    private Shop queryWithLogicalExpire(Long id){
        // 获取 key
        String key  = CACHE_SHOP_KEY + id;

        // 查询缓存
        String json = stringRedisTemplate.opsForValue().get(key);

        // 缓存未命中, 或命中空值, 返回空值 (缓存未命中, 查询数据库并写入 Redis 的逻辑放在缓存穿透方法处理)
        if(StrUtil.isBlank(json)){
            return null;
        }

        // 缓存命中, 反序列化两次结果, 拿到 Shop 对象
        RedisData redisData = JSONUtil.toBean(json, RedisData.class);
        LocalDateTime expireTime = redisData.getExpireTime();
        // Object -> JSONObject -> Shop
        Shop shop = JSONUtil.toBean((JSONObject) redisData.getData(), Shop.class);

        // 判断逻辑是否过期
        if(expireTime.isAfter(LocalDateTime.now())){
            // 未过期, 直接返回
            return shop;
        }

        // 逻辑过期, 获取互斥锁
        String lockKey = LOCK_SHOP_KEY + id;
        boolean isLock = this.getLock(lockKey);

        // 获取成功, 异步执行缓存重建任务, 并返回旧数据
        if(isLock){
            CACHE_REBUILD_EXECUTOR.submit( () -> {
                try {
                    // 缓存重建 (查询数据库, 写入缓存, 设置逻辑过期时间)
                    this.saveShop2Redis(id, 20L);
                } catch (Exception e) {
                    throw new RuntimeException(e);
                } finally {
                    // 释放互斥锁
                    this.delLock(lockKey);
                }
            });
        }

        // 获取失败, 返回旧数据
        return shop;
    }
}




